package com.frejoys.service.impl;


import cn.afterturn.easypoi.cache.manager.IFileLoader;
import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollectionUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.frejoys.common.dto.admin.SetAgentDto;
import com.frejoys.common.dto.admin.StoreIdAndPageDto;
import com.frejoys.common.dto.admin.StoreIdAndPhoneDto;
import com.frejoys.common.dto.app.AgentStoreIdDto;
import com.frejoys.common.dto.common.PageDto;
import com.frejoys.common.enums.EFileSize;
import com.frejoys.common.enums.EStatus;
import com.frejoys.common.enums.EUserRole;
import com.frejoys.common.error.HttpStatus;
import com.frejoys.common.exception.GeneralException;
import com.frejoys.common.util.AppConfigUtil;
import com.frejoys.common.util.CommonUtil;
import com.frejoys.common.util.TimeUtil;
import com.frejoys.common.vo.admin.AdminAgentStoreListVo;
import com.frejoys.common.vo.admin.AdminAgentStoreVo;
import com.frejoys.common.vo.admin.StoreAndeAgentAndUserListVo;
import com.frejoys.common.vo.agent.AgentStoreVo;
import com.frejoys.common.vo.agent.AgentUserInfoVo;
import com.frejoys.common.vo.app.AgentStoreInfoVo;
import com.frejoys.common.vo.app.AgentStoreListVo;
import com.frejoys.common.vo.store.StoreUserVo;
import com.frejoys.dao.entity.*;
import com.frejoys.dao.mapper.*;
import com.frejoys.service.*;
import lombok.RequiredArgsConstructor;
import org.checkerframework.checker.units.qual.A;
import org.springframework.stereotype.Service;
import org.springframework.util.ObjectUtils;

import java.math.BigDecimal;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

@Service
@RequiredArgsConstructor
public class AgentStoreServiceImpl extends ServiceImpl<AgentStoreMapper, AgentStore> implements AgentStoreService {

    private final UserMapper userMapper;

    private final UserService userService;

    private final StoreMapper storeMapper;

//    private final StoreService storeService;

    private final CategoryService categoryService;

    private final StoreUserMapper storeUserMapper;

    private final AreaService areaService;

    private final UserLevelStoreMoneyMapper userLevelStoreMoneyMapper;

    @Override
    public boolean addInfo(Long agentUserId, Long userId, Integer storeId) {
        return save(new AgentStore() {{
            setAgentUserId(agentUserId);
            setUserId(userId);
            setStoreId(storeId);
            setCreateTime(TimeUtil.unixTime());
        }});
    }

    @Override
    public boolean delInfo(Integer storeId) {
        LambdaUpdateWrapper<AgentStore> wrapper = new LambdaUpdateWrapper<>();
        wrapper.set(AgentStore::getStatus, EStatus.off.getValue());
        wrapper.eq(AgentStore::getStoreId, storeId);
        return update(wrapper);
    }

    @Override
    public Long getAgentUserIdByStoreId(Integer storeId) {
        LambdaQueryWrapper<AgentStore> wrapper = new LambdaQueryWrapper<>();
        wrapper.select(AgentStore::getAgentUserId).eq(AgentStore::getStoreId, storeId);
        AgentStore e = getOnly(wrapper);
        return e != null ? e.getAgentUserId() : null;
    }

    /**
     * 根据代理人id获取代理店铺列表
     *
     * @param userId
     */
    @Override
    public List<AgentStoreListVo> listAgentStoreByUserId(PageDto dto,Long userId) {
        Page page = new Page<>(dto.getPage(), dto.getSize());
        page.setSearchCount(false);

        //查询代理人信息
        User user = userService.getInfo(userId);
        if (ObjectUtils.isEmpty(user)) {
            throw new GeneralException(HttpStatus.userNoExist);
        }
        if (!CommonUtil.bitExistOffset(user.getRole(), EUserRole.agent.getValue())) {
            throw new GeneralException(HttpStatus.agentRoleNoExist);
        }
        if (user.getStatus().equals(EStatus.del.getValue())) {
            //代理人被禁止登录
            throw new GeneralException(HttpStatus.userStatusLoginError);
        }

        LambdaQueryWrapper<AgentStore> wrapper = Wrappers.lambdaQuery();
        wrapper.eq(AgentStore::getAgentUserId, userId)
                .eq(AgentStore::getStatus, EStatus.on.getValue())
                .orderByDesc(AgentStore::getCreateTime);

        List<AgentStore> agentStores = baseMapper.selectPage(page,wrapper).getRecords();
        List<AgentStoreListVo> list = new ArrayList<>();

        if (!CollectionUtil.isEmpty(agentStores)) {

            List<Integer> storeIds = agentStores.stream().map(AgentStore::getStoreId).toList();
            List<Long> userIds = agentStores.stream().map(AgentStore::getUserId).toList();

            //店铺信息
            List<Store> storeList = storeMapper.getStoreList(storeIds);
            Map<Integer, Store> storeMap = storeList.stream().collect(Collectors.toMap(Store::getId, store -> store));

            //商家用户信息
            List<User> userList = userMapper.getUserList(userIds, EStatus.on.getValue());
            Map<Long, User> userMap = userList.stream().collect(Collectors.toMap(User::getId,user1 -> user1));

            //查询推广金额
//            List<UserLevelStoreMoney> userLevelMonies = userLevelStoreMoneyMapper.storeMoneyByUserId(userId);
//            Map<Integer, BigDecimal> childUserMoneyMap = userLevelMonies.stream().collect(Collectors.toMap(UserLevelStoreMoney::getStoreId, UserLevelStoreMoney::getMoney));

            agentStores.forEach(agentStore -> {

                AgentStoreListVo vo = new AgentStoreListVo();
                vo.setId(agentStore.getId());
                vo.setCreateTime(agentStore.getCreateTime());

                if (storeMap.containsKey(agentStore.getStoreId())){
                    Store store = storeMap.get(agentStore.getStoreId());
                    vo.setName(store.getName());
                    vo.setLogo(AppConfigUtil.getUrlPrefix(store.getLogo(), EFileSize.logo.getValue()));
                }

                if (userMap.containsKey(agentStore.getUserId())){
                    User u = userMap.get(agentStore.getUserId());
                    vo.setNickname(u.getNickname());
                    vo.setRole(u.getRole());
                    vo.setAvatar(AppConfigUtil.getUrlPrefix(u.getAvatar()));
                }
//                if (childUserMoneyMap.containsKey(agentStore.getStoreId())){
//                    vo.setChildUserMoney(childUserMoneyMap.get(agentStore.getStoreId()));
//                }
                list.add(vo);
            });
        }
        return list;
    }


    @Override
    public List<AdminAgentStoreListVo> listAgentStore() {
        LambdaQueryWrapper<AgentStore> wrapper = Wrappers.lambdaQuery();
        wrapper.select(AgentStore::getStoreId).gt(AgentStore::getStatus, EStatus.del.getValue());
        List<AgentStore> agentStores = baseMapper.selectList(wrapper);
        if (CollectionUtil.isEmpty(agentStores)) {
            return Collections.emptyList();
        }
        List<Integer> storeIds = agentStores.stream().map(AgentStore::getStoreId).toList();

        LambdaQueryWrapper<Store> storeWrapper = Wrappers.lambdaQuery();
        storeWrapper.select(Store::getId, Store::getName, Store::getCoverShow, Store::getTel).in(Store::getId, storeIds);
        List<Store> stores = storeMapper.selectList(storeWrapper);
        List<AdminAgentStoreListVo> list = BeanUtil.copyToList(stores, AdminAgentStoreListVo.class);
        if (CollectionUtil.isNotEmpty(list)) {
            for (AdminAgentStoreListVo a : list) {
                a.setCoverShow(AppConfigUtil.getUrlPrefix(a.getCoverShow()));
            }
        }
        return list;
    }

    /**
     * 后台获取代理店铺详情
     *
     * @param storeId 店铺id
     */
    @Override
    public AdminAgentStoreVo getStoreInfo(Integer storeId) {


        AgentUserInfoVo agentUserVo = storeUserMapper.getStoreAgentUserByStoreId(storeId);

        AgentStoreVo agentStoreVo = storeMapper.getStoreInfo(storeId);

        if (ObjectUtil.isNotEmpty(agentStoreVo)) {
            AdminAgentStoreVo adminAgentStoreVo = BeanUtil.copyProperties(agentStoreVo, AdminAgentStoreVo.class);

            StoreUserVo storeUser = storeUserMapper.getStoreUserByStoreId(storeId);

            Map<Integer, String> map = categoryService.getIdNameMap();
            adminAgentStoreVo.setCover(AppConfigUtil.getUrlPrefix(adminAgentStoreVo.getCover()));
            adminAgentStoreVo.setLogo(AppConfigUtil.getUrlPrefix(adminAgentStoreVo.getLogo(),EFileSize.logo.getValue()));
            adminAgentStoreVo.setCoverShow(AppConfigUtil.getUrlPrefix(adminAgentStoreVo.getCoverShow()));
            adminAgentStoreVo.setCategoryName(map.get(adminAgentStoreVo.getCategoryId()));
            adminAgentStoreVo.setUserVo(storeUser);
            agentUserVo.setAvatar(AppConfigUtil.getUrlPrefix(agentUserVo.getAvatar()));
            adminAgentStoreVo.setAgentUser(agentUserVo);
            return adminAgentStoreVo;
        }
        return new AdminAgentStoreVo();
    }



    //修改代理人
    @Override
    public Integer setAgent(SetAgentDto dto) {
        LambdaUpdateWrapper<AgentStore> updateWrapper = new LambdaUpdateWrapper<>();
        updateWrapper.eq(AgentStore::getId,dto.getId());
        updateWrapper.set(AgentStore::getAgentUserId, dto.getAgentUserId());
        return baseMapper.update(updateWrapper);
    }



    @Override
    public AgentStoreInfoVo agentStoreInfo(Integer id) {
        LambdaQueryWrapper<AgentStore> wrapper = new LambdaQueryWrapper<>();
        wrapper.select(AgentStore::getUserId,AgentStore::getStoreId)
                .eq(AgentStore::getId,id);

        AgentStore agentStore = getOnly(wrapper);
        AgentStoreInfoVo vo = new AgentStoreInfoVo();

        if (agentStore != null){

            LambdaQueryWrapper<User> userQueryWrapper = new LambdaQueryWrapper<>();
            userQueryWrapper.select(User::getNickname,User::getAvatar,User::getPhone)
                    .eq(User::getId,agentStore.getUserId())
                    .eq(User::getStatus,EStatus.on.getValue());

            User user = userService.getOnly(userQueryWrapper);

            if (user == null){
                throw new GeneralException(HttpStatus.userNoExist);
            }

            LambdaQueryWrapper<Store> storeQueryWrapper = new LambdaQueryWrapper<>();
            storeQueryWrapper.eq(Store::getId,agentStore.getStoreId());

            AgentStoreVo store = storeMapper.getStoreInfo(agentStore.getStoreId());

            if (store == null){
                throw new GeneralException(HttpStatus.agentStoreNoExist);
            }
            Map<String, String> areaMap = areaService.getNameMap();

            vo.setCityName(areaMap.get(store.getCity()));
            vo.setDistrictName(areaMap.get(store.getDistrict()));
            vo.setProvinceName(areaMap.get(store.getProvince()));
            vo.setTel(store.getTel());
            vo.setAddress(store.getAddress());
            vo.setName(store.getName());
            vo.setEndTime(store.getEndTime());
            vo.setStartTime(store.getStartTime());
            vo.setStreet(store.getStreet());
            vo.setShelves(store.getShelves());
            vo.setCover(AppConfigUtil.getUrlPrefix(store.getCover()));
            vo.setLogo(AppConfigUtil.getUrlPrefix(store.getLogo(),EFileSize.logo.getValue()));

            vo.setNickname(user.getNickname());
            vo.setAvatar(AppConfigUtil.getUrlPrefix(user.getAvatar()));
            vo.setPhone(user.getPhone());
        }

        return vo;
    }

    @Override
    public IPage<StoreAndeAgentAndUserListVo> storeAndeAgentAndUserListInfo(StoreIdAndPhoneDto dto) {
        Page page = new Page<>(dto.getPage(), dto.getSize());

        LambdaQueryWrapper<AgentStore> wrapper = new LambdaQueryWrapper<>();
        if (dto.getStoreId() != null){
            wrapper.eq(AgentStore::getStoreId,dto.getStoreId());
        }
        if (StrUtil.isNotEmpty(dto.getPhone())){
            Long userId = userMapper.phoneByUserId(dto.getPhone());
            if (userId == null){
                return new Page<>();
            }
            wrapper.eq(AgentStore::getUserId,userId);
        }
        if (StrUtil.isNotEmpty(dto.getAgentPhone())){
            Long userId = userMapper.phoneByUserId(dto.getAgentPhone());
            if (userId == null){
                return new Page<>();
            }
            wrapper.eq(AgentStore::getAgentUserId,userId);
        }
        if (StrUtil.isNotEmpty(dto.getName())){
            List<Store> stores = storeMapper.getStoreNameById(dto.getName());
            if (stores.isEmpty()){
                return new Page<>();
            }
            List<Integer> ids = stores.stream().map(Store::getId).toList();
            wrapper.in(AgentStore::getStoreId,ids);
        }
        wrapper.orderByDesc(AgentStore::getCreateTime);
        Page<AgentStore> agentStorePage = baseMapper.selectPage(page, wrapper);
        IPage<StoreAndeAgentAndUserListVo> agentStoreIPage = agentStorePage.convert(agentStore -> BeanUtil.copyProperties(agentStore, StoreAndeAgentAndUserListVo.class));
        List<StoreAndeAgentAndUserListVo> list = agentStoreIPage.getRecords();

        if (!list.isEmpty()){

            List<Integer> storeIds = list.stream().map(StoreAndeAgentAndUserListVo::getStoreId).toList();
            List<Long> agentUserIds = list.stream().map(StoreAndeAgentAndUserListVo::getAgentUserId).toList();
            List<Long> userIds = list.stream().map(StoreAndeAgentAndUserListVo::getUserId).toList();

            //查询店铺信息
            LambdaQueryWrapper<Store> storeWrapper = new LambdaQueryWrapper<>();
            storeWrapper.select(Store::getName,Store::getLogo,Store::getId).in(Store::getId,storeIds);
            List<Store> storeList = storeMapper.selectList(storeWrapper);
            Map<Integer, Store> storeMap = storeList.stream().collect(Collectors.toMap(Store::getId, store -> store));

            //查询代理人信息
            List<User> agentUserList = userMapper.getUserByIds(agentUserIds);
            Map<Long, User> agentUserMap = agentUserList.stream().collect(Collectors.toMap(User::getId, user -> user));

            //查询店铺拥有者信息
            List<User> userList = userMapper.getUserByIds(userIds);
            Map<Long, User> userMap = userList.stream().collect(Collectors.toMap(User::getId, user -> user));

            list.forEach(vo ->{
                //插入店铺信息
                Store store = storeMap.get(vo.getStoreId());
                if (!ObjectUtil.isEmpty(store)){
                    vo.setStoreName(store.getName());
                    vo.setLogo(AppConfigUtil.getUrlPrefix(store.getLogo(),EFileSize.logo.getValue()));
                }

                //插入代理人信息
                User agentUser = agentUserMap.get(vo.getAgentUserId());
                if (!ObjectUtil.isEmpty(agentUser)){
                    vo.setAgentNickname(agentUser.getNickname());
                    vo.setAgentPhone(agentUser.getPhone());
                }

                //插入店铺拥有人信息
                User user = userMap.get(vo.getUserId());
                if (!ObjectUtil.isEmpty(user)){
                    vo.setNickname(user.getNickname());
                    vo.setPhone(user.getPhone());
                }
            });
        }
        return agentStoreIPage;
    }

}
